#pragma once

#include <iostream>
#include <memory>
#include <string>
#include <vector>
#include <queue>
#include <unistd.h>
#include "Cond.hpp"
#include "Mutex.hpp"
#include "Log.hpp"
#include "Thread.hpp"

namespace ThreadPoolModule
{

    using namespace CondModule;
    using namespace MutexModule;
    using namespace LogModule;
    using namespace ThreadModule;

    using thread_t = std::shared_ptr<Thread>;

    const static int defaultthreadnum = 5;

    void Default()
    {
        LOG(LogLevel::DEBUG) << "我是一个测试函数";
    }

    template <typename T>
    class ThreadPool
    {
    private:
        bool IsEmpty() { return _taskq.empty(); }
        void HandlerTask(std::string name)
        {
            LOG(LogLevel::INFO) << "线程: " << name << ", 进入HandlerTask的逻辑";
            while (true)
            {
                T t;
                {
                    // 1.拿任务
                    MutexGuard mutexguard(_mutex);
                    while (IsEmpty() && _isrunning)
                    {
                        // 当线程池在运行状态且任务队列不为空，线程才需要去等待
                        _waitnum++;
                        _cond.Wait(_mutex);
                        _waitnum--;
                    }
                    if (IsEmpty() && !_isrunning)
                    {
                        // 如果任务队列为空且线程池退出了，就不用继续执行任务了
                        break;
                    }
                    t = _taskq.front();
                    _taskq.pop();
                }
                // 2.处理任务:不需要加锁处理
                t(name);
            }
            LOG(LogLevel::INFO) << "线程:" << name << "退出";
        }

    public:
        ThreadPool(int num = defaultthreadnum)
            : _threadnum(num), _waitnum(0), _isrunning(false)
        {
            // 构建线程
            for (int i = 0; i < _threadnum; i++)
            {
                _threads.push_back(std::make_shared<Thread>(std::bind(&ThreadPool::HandlerTask, this, std::placeholders::_1)));
                LOG(LogLevel::INFO) << "构建线程" << _threads.back()->Name();
            }
        }
        void Start()
        {
            // 启动线程池
            if (_isrunning)
                return; // 如果线程池已经是启动的，就不要重复启动了
            _isrunning = true;
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Start();
                LOG(LogLevel::INFO) << "启动线程" << thread_ptr->Name();
            }
        }
        void Enqueue(T &&in)
        {
            MutexGuard mutexguard(_mutex);
            if (!_isrunning)
                return; // 进程池退了，就不能让新任务入队列了
            _taskq.push(std::move(in));
            if (_waitnum > 0)
            {
                _cond.Notify();
            }
        }
        void Wait()
        {
            // 等待回收线程Wait
            for (auto &thread_ptr : _threads)
            {
                thread_ptr->Join();
                LOG(LogLevel::INFO) << "回收线程" << thread_ptr->Name();
            }
        }
        void Stop()
        {
            // 停止线程池：1.让线程自己停下来，避免在执行任务中途被中断；2.线程池一旦停止，就不能让新任务继续进来了；3.历史任务必须全部执行完毕
            MutexGuard mutexguard(_mutex);
            if (_isrunning)
            {
                _isrunning = false;
                if (_waitnum > 0)
                {
                    _cond.NotifyAll(); // 如果有任务，线程就不会去等待；如果需要唤醒，说明此时任务队列为空了
                }
            }
        }
        ~ThreadPool()
        {
        }

    private:
        std::vector<thread_t> _threads;
        int _threadnum; // 线程的数量
        std::queue<T> _taskq;
        int _waitnum; // 等待队列的线程
        Mutex _mutex;
        Cond _cond;
        bool _isrunning; // 判断线程池是否启动
    };
}